import os
from app.config import config

def init_directories():
    """初始化数据目录"""
    os.makedirs(config.DATA_DIR, exist_ok=True)
    print(f"创建数据目录: {config.DATA_DIR}")

def init_env():
    """初始化环境变量文件"""
    if not os.path.exists('.env'):
        with open('.env', 'w') as f:
            f.write("""# 存储配置
DATA_DIR=data
MAX_RECORDS=1000""")
        print("创建 .env 文件")

def main():
    """主初始化函数"""
    print("开始初始化服务器...")
    
    # 初始化环境变量
    init_env()
    
    # 初始化目录
    init_directories()
    
    print("\n初始化完成！")

if __name__ == '__main__':
    main()